import pandas as pd

# 读取 summerOly_medal_counts.csv 文件，获取 NOC 和 Year 列
medal_df = pd.read_csv('summerOly_medal_counts.csv')

# 过滤出 2020 年的数据
medal_2020_df = medal_df[medal_df['Year'] == 2020]

# 提取得过奖的国家（NOC 列）
medal_countries_2020 = medal_2020_df['NOC'].dropna().unique()

# 读取 tokyo_olympic_participating_countries_2020.csv 文件，获取奥运参赛国家（Country 列）
olympic_df = pd.read_csv('tokyo_olympic_participating_countries_2020.csv')

# 筛选出没有在 2020 年得过奖的国家
non_medal_countries_2020 = [country for country in olympic_df['Country'] if country not in medal_countries_2020]

# 将没有得奖的国家保存为一个新的 DataFrame
non_medal_df_2020 = pd.DataFrame(non_medal_countries_2020, columns=["Country"])

# 保存为新的 CSV 文件
non_medal_df_2020.to_csv('non_medal_countries_2020.csv', index=False, encoding='utf-8')

print("没有获奖的2020年国家已保存为 non_medal_countries_2020.csv")
